export dataset_name=/home/user4/workspace/ddg-benchmark/data

# mifst mif carp_640M
CUDA_VISIBLE_DEVICES=2 python src/MIF-ST/predict.py \
    --dataset_input $dataset_name \
    --model_location mifst mif